from train_models.hmm import process_hzpy, process_article, gen_base, process_finally

def train_base():
    process_hzpy.process_hzpy()
    process_article.gen_sentence()
    gen_base.gen_base()
    # process_finally.gen_finally()

def train_pretrained(times):
    # 多次训练
    process_hzpy.process_hzpy()
    process_article.gen_sentence()
    for i in range(times):
        gen_base.gen_pretrained()
        # process_finally.gen_finally()

if __name__ == '__main__':
    train_base()